Python get the index of maximum value in a list of lists